class WorkloadManager:
    workloads_dict = {}
    @classmethod
    def register_workload(cls, workload_name):
        def wrapper(workload_class):
            cls.workloads_dict.update({workload_name: workload_class})
            return workload_class
        return wrapper

    @classmethod
    def get_workload_class(cls, workload_name):
        return cls.workloads_dict.get(workload_name)

    @classmethod
    def is_registered(cls, workload_name):
        if workload_name in cls.workloads_dict:
            return True
        else:
            return False
